/* Copyright (c) Huawei Technologies Co., Ltd. 2025. All rights reserved.

Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at

        http://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
        limitations under the License.
==============================================================================*/

#ifndef TILING_POLICY_DEFINE_H
#define TILING_POLICY_DEFINE_H

namespace HstuDenseForwardFuxi {

#include <cstdio>
#include <cstdint>
#include "register/tilingdata_base.h"
#include "tiling/tiling_api.h"

namespace INDEX_T {
    constexpr int INDEX_0 = 0;
    constexpr int INDEX_1 = 1;
    constexpr int INDEX_2 = 2;
    constexpr int INDEX_3 = 3;
    constexpr int INDEX_4 = 4;
    constexpr int INDEX_5 = 5;
}

constexpr int FLOAT16_TILING_KEY = 0;
constexpr int TRANS_TASK_NUM = 3;
#if defined(__DAV_C310_VEC__)
    constexpr int MAX_AIV_NUM = 64;
#else
    constexpr int MAX_AIV_NUM = 48;
#endif

#ifdef SUPPORT_V200
    constexpr int OUTPUT_DIM_NUM = 3;
    constexpr int MAX_BATCH_SIZE = 10;
    constexpr int BLOCK_HEIGHT = 64;
    constexpr int VCORE_NUM_IN_ONE_AIC = 1;
    constexpr int COMPUTE_PIPE_NUM = 3;
    constexpr int TRANS_PIPE_NUM = 1;
#else
    constexpr int OUTPUT_DIM_NUM = 2;
    constexpr int MAX_BATCH_SIZE = 512;
    constexpr int BLOCK_HEIGHT = 256;
    constexpr int VCORE_NUM_IN_ONE_AIC = 2;
    constexpr int COMPUTE_PIPE_NUM = 3;
    constexpr int TRANS_PIPE_NUM = 4;
    constexpr int OUTPUT_DIM2_TIMES_3 = 3;
#endif

}

#endif